# look node and variable's name


def look_name():
    from tensorflow.python import pywrap_tensorflow
    reader = pywrap_tensorflow.NewCheckpointReader('model.ckpt')
    var_to_shape_map = reader.get_variable_to_shape_map()
    for key in var_to_shape_map.keys():
        print("tensor_name: ", key)
